package com.zzg.algorithm.floyd.entity;

import java.util.Arrays;

/**
 * @Author zhengzg
 * @Date 2022-10-10
 * @Version v1.0
 */
public class Graph {
    // 存放顶点数组
    private char[] vertex;
    // 保存，从各个顶点出发到其他顶点的巨鹿，最后的记过，保留在该数组
    private int[][] dis;
    // 保存到达目标顶点的前驱结点
    private int[][] pre;

    public Graph(int length,int[][] matrix,char[] vertex) {
        this.vertex = vertex;
        this.dis = matrix;
        this.pre = new int[length][length];

        // 对pre数组初始化，注意存放的是前驱顶点的下标
        for (int i = 0;i < length;i++) {
            Arrays.fill(pre[i],i);
        }
    }

    // 显示pre数组和dis数组
    public void show() {
        char[] vertex = { 'A', 'B', 'C', 'D', 'E', 'F', 'G' };
        for (int k = 0;k < dis.length;k++) {
            // 先将pre数组输出的一行
            for (int i = 0; i < dis.length; i++) {
                System.out.print(vertex[pre[k][i]] + " ");
            }
            System.out.println();
            // 输出dis数组的一行数据
            for (int i = 0; i < dis.length; i++) {
                System.out.print("(" + vertex[k] + "到" + vertex[i] + "最短路径是" + dis[k][i] + ")");
            }
            System.out.println();
            System.out.println();
        }
    }

    // 弗洛伊德算法
    public void floyd() {
        // 保存距离
        int len = 0;
        // 从中间顶点开始遍历，k就是中间顶点的下标[A,B,C,D,E,F,G]
        for (int k = 0; k < dis.length; k++) {
            // 从i顶点开始出发
            for (int i = 0; i < dis.length; i++) {
                // 到达j顶点
                for (int j = 0; j < dis.length; j++) {
                    // 求出从 i 顶点出发，经过 k 中间顶点，到达 j 顶点距离
                    len = dis[i][k] + dis[k][j];
                    if (len < dis[i][j]) {
                        // 更新距离
                        dis[i][j] = len;
                        // 更新前驱顶点
                        pre[i][j] = pre[k][j];
                    }
                }
            }
        }
    }
}
